package cn.icatw.leetcode.editor.cn;
//给你一个整数数组 nums 和一个整数 k ，请你统计并返回 该数组中和为 k 的子数组的个数 。
//
// 子数组是数组中元素的连续非空序列。
//
//
//
// 示例 1：
//
//
//输入：nums = [1,1,1], k = 2
//输出：2
//
//
// 示例 2：
//
//
//输入：nums = [1,2,3], k = 3
//输出：2
//
//
//
//
// 提示：
//
//
// 1 <= nums.length <= 2 * 10⁴
// -1000 <= nums[i] <= 1000
// -10⁷ <= k <= 10⁷
//
//
// Related Topics 数组 哈希表 前缀和 👍 2431 👎 0


import java.util.HashMap;

//Java：和为 K 的子数组
public class T560_SubarraySumEqualsK {
    public static void main(String[] args) {
        Solution solution = new T560_SubarraySumEqualsK().new Solution();
        // TO TEST
        solution.subarraySum(new int[]{1, 1, 1}, 2);
    }

    //leetcode submit region begin(Prohibit modification and deletion)
    class Solution {
        public int subarraySum(int[] nums, int k) {
            // 用于记录前缀和出现的次数
            HashMap<Integer, Integer> prefixSumMap = new HashMap<>();
            // 初始化前缀和为0的情况（相当于一个空数组的前缀和）
            prefixSumMap.put(0, 1);

            int currentSum = 0;  // 当前前缀和
            int count = 0;  // 记录满足条件的子数组个数

            // 遍历数组
            for (int num : nums) {
                currentSum += num;  // 更新当前前缀和

                // 检查是否存在满足条件的前缀和
                if (prefixSumMap.containsKey(currentSum - k)) {
                    count += prefixSumMap.get(currentSum - k);
                }

                // 更新哈希表中的前缀和出现次数
                prefixSumMap.put(currentSum, prefixSumMap.getOrDefault(currentSum, 0) + 1);
            }

            return count;
        }
    }
//leetcode submit region end(Prohibit modification and deletion)

}
